/**
 * 
 */
package com.ubo.graphviz.model.graph.statement.node.classNode;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import com.ubo.graphviz.model.graph.statement.node.NodeStmt;
import com.ubo.visu.model.EEntityType;

/**
 * @author saludegu
 * 
 */
public class ClassStmt extends NodeStmt {

	/**
	 * Elements contenus dans la classe i.e. Attributs ou Methodes.
	 */
	protected List<IClassElement> mListElements;

	/**
	 * Type de l'entité.
	 */
	protected EEntityType mEntityType;

	/**
	 * Constructor
	 * 
	 * @param nodeName
	 */
	public ClassStmt(String nodeName, int idClass, EEntityType entityType) {
		super(nodeName, idClass, entityType);
	}

	/**
	 * Ajout d'un element à la liste.
	 * 
	 * @param newElement
	 */
	public void addElement(IClassElement newElement) {
		if (mListElements == null)
			mListElements = new ArrayList<IClassElement>();
		mListElements.add(newElement);
	}

	/**
	 * Getter des elements contenus dans la classe.
	 * 
	 * @return
	 */
	public Collection<IClassElement> getElements() {
		return mListElements;
	}

}
